package top.minuy.subject.leetcode._752;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.HashSet;

/**
 * 752. 打开转盘锁
 * 你有一个带有四个圆形拨轮的转盘锁。每个拨轮都有10个数字： '0', '1', '2', '3', '4', '5', '6', '7', '8', '9' 。每个拨轮可以自由旋转：例如把 '9' 变为 '0'，'0' 变为 '9' 。每次旋转都只能旋转一个拨轮的一位数字。
 * <p>
 * 锁的初始数字为 '0000' ，一个代表四个拨轮的数字的字符串。
 * <p>
 * 列表 deadends 包含了一组死亡数字，一旦拨轮的数字和列表里的任何一个元素相同，这个锁将会被永久锁定，无法再被旋转。
 * <p>
 * 字符串 target 代表可以解锁的数字，你需要给出解锁需要的最小旋转次数，如果无论如何不能解锁，返回 -1 。
 * <p>
 * <p>
 * <p>
 * 示例 1:
 * <p>
 * 输入：deadends = ["0201","0101","0102","1212","2002"], target = "0202"
 * 输出：6
 * 解释：
 * 可能的移动序列为 "0000" -> "1000" -> "1100" -> "1200" -> "1201" -> "1202" -> "0202"。
 * 注意 "0000" -> "0001" -> "0002" -> "0102" -> "0202" 这样的序列是不能解锁的，
 * 因为当拨动到 "0102" 时这个锁就会被锁定。
 * 示例 2:
 * <p>
 * 输入: deadends = ["8888"], target = "0009"
 * 输出：1
 * 解释：
 * 把最后一位反向旋转一次即可 "0000" -> "0009"。
 * 示例 3:
 * <p>
 * 输入: deadends = ["8887","8889","8878","8898","8788","8988","7888","9888"], target = "8888"
 * 输出：-1
 * 解释：
 * 无法旋转到目标数字且不被锁定。
 * 示例 4:
 * <p>
 * 输入: deadends = ["0000"], target = "8888"
 * 输出：-1
 * <p>
 * <p>
 * 提示：
 * <p>
 * 1 <= deadends.length <= 500
 * deadends[i].length == 4
 * target.length == 4
 * target 不在 deadends 之中
 * target 和 deadends[i] 仅由若干位数字组成
 * 通过次数72,911提交次数137,524
 *
 * @author Minuy
 * @time 20:14
 * @date 2021/11/20
 */
public class Solution {
    String[] deadends;

    public int openLock(String[] deadends, String target) {
        this.deadends = deadends;

        HashSet<String> isVisited = new HashSet<>();
        HashMap<String, Integer> dis = new HashMap<>();

        String sta = "0000";

        if (notDead(sta)) { // 最开始的这个也要判断是不是凉凉的


            ArrayList<String> queue = new ArrayList<>();

            queue.add(sta);
            isVisited.add(sta);
            dis.put(sta, 0);

            if(sta.equals(target)){
                return dis.get(target);
            }

            while (!queue.isEmpty()) {
                sta = queue.remove(0);

                for (String s : getNextStats(sta)) {
                    if (!isVisited.contains(s)) {
                        queue.add(s);
                        isVisited.add(s);
                        dis.put(s, dis.get(sta) + 1);
                        if (s.equals(target)) {
                            return dis.get(target);
                        }
                    }
                }
            }
        }
        return -1;
    }

    // 获取与sta相邻的状态
    private Iterable<String> getNextStats(String sta) {
        ArrayList<String> ret = new ArrayList<>();
        for (int i = 0; i < sta.length(); i++) {

            char[] charsAdd = sta.toCharArray();
            char[] charsMum = sta.toCharArray();

            if (charsAdd[i] == '9') {
                charsAdd[i] = '0';
            } else {
                charsAdd[i] += 1;
            }
            if (charsMum[i] == '0') {
                charsMum[i] = '9';
            } else {
                charsMum[i] -= 1;
            }
            String a = new String(charsAdd);
            if (notDead(a)) {
                ret.add(a);
            }

            String m = new String(charsMum);
            if (notDead(m)) {
                ret.add(m);
            }
        }
        return ret;
    }

    // 判断有没有触碰到死亡密码
    private boolean notDead(String str) {
        for (String d : deadends) {
            if (str.equals(d)) {
                return false;
            }
        }
        return true;
    }

    public static void main(String[] args) {
        String s = "0000";
        Solution solution = new Solution();
        String[] d = {"0201", "0101", "0102", "1212", "2002"};
        System.out.println(solution.openLock(d, "0202"));
        System.out.println(solution.getNextStats(s));
    }
}
